cmake_minimum_required(VERSION 3.10)

project(VibeCUDA VERSION 1.0)

find_package(OpenCV REQUIRED)
find_package(CUDA REQUIRED)

enable_language(CUDA)

set(SOURCES app.cpp vibe3.cu vibe3.h)

include_directories(${OpenCV_INCLUDE_DIRS} ${CUDA_INCLUDE_DIRS})

set(CMAKE_CUDA_STANDARD 14)
set(CMAKE_CUDA_STANDARD_REQUIRED ON)

# 设置 CUDA 架构
set(CMAKE_CUDA_ARCHITECTURES 86)  # 例如，对于 RTX 3000 系列 GPU

cuda_add_executable(main ${SOURCES})

target_link_libraries(main ${OpenCV_LIBS} ${CUDA_LIBRARIES})
